package com.xhorse.lockplatform.comm.util;

import sun.misc.BASE64Decoder;

import javax.crypto.Cipher;
import javax.crypto.spec.SecretKeySpec;
import java.io.FileOutputStream;
import java.io.IOException;
import java.security.MessageDigest;

public class DecriptUtil {

    /**
     * AES加密
     *
     * @param str
     * @param str：待加密的字符串，secretKeyBase：用于生成密匙的基础字符串
     * @return 加密字节数组
     */
    public static byte[] encryptAES(String str, String secretKeyBase) {
        try {
            //5.根据字节数组生成AES密匙
            SecretKeySpec key = new SecretKeySpec(secretKeyBase.getBytes(), "AES");
            //6.创建密码器
            Cipher cipher = Cipher.getInstance("AES/ECB/PKCS5Padding");
            byte[] byteContent = str.getBytes("UTF-8");
            //7.初始化密码器，第一个参数为加密(Encrypt_mode)或者解密(Decrypt_mode)操作，第二个参数为使用的KEY
            cipher.init(Cipher.ENCRYPT_MODE, key);
            byte[] result = cipher.doFinal(byteContent);
            return result; // 加密
        } catch (Exception e) {
            throw new RuntimeException("加密字符串[" + str + "]时遇到异常", e);
        }
    }

    /**
     * AES解密
     *
     * @param strByteArray：待解密的字节数组
     * @param secretKeyBase：由于AES是对称加密，所以secreKeybase在加密解密时要一样
     * @return 解密后的字符串
     */
    public static String decryptAES(byte[] strByteArray, String secretKeyBase) {
        try {
            SecretKeySpec key = new SecretKeySpec(secretKeyBase.getBytes(), "AES");
            Cipher cipher = Cipher.getInstance("AES/ECB/NoPadding");// 创建密码器
            cipher.init(Cipher.DECRYPT_MODE, key);// 初始化
            String result = new String(cipher.doFinal(strByteArray), "utf-8");
            return result; // 加密
        } catch (Exception e) {
            throw new RuntimeException("解密字节[" + strByteArray + "]时遇到异常", e);
        }
    }

    /**
     * SHA1加密
     *
     * @param str 待加密的字符串
     * @return 加密串
     */
    public static String SHA1(String str) {
        try {
            MessageDigest digest = MessageDigest
                    .getInstance("SHA-1"); //如果是SHA加密只需要将"SHA-1"改成"SHA"即可
            digest.update(str.getBytes());
            byte messageDigest[] = digest.digest();
            // Create Hex String
            StringBuffer hexStr = new StringBuffer();
            // 字节数组转换为 十六进制 数
            for (int i = 0; i < messageDigest.length; i++) {
                String shaHex = Integer.toHexString(messageDigest[i] & 0xFF);
                if (shaHex.length() < 2) {
                    hexStr.append(0);
                }
                hexStr.append(shaHex);
            }
            return hexStr.toString();

        } catch (Exception e) {
            throw new RuntimeException("加密字符串[" + str + "]时遇到异常", e);
        }
    }

    /**
     * MD5加密
     *
     * @param str：待加密的字符串
     * @return 加密串
     */
    public static String MD5(String str) {
        try {
            // 获得MD5摘要算法的 MessageDigest 对象
            MessageDigest mdInst = MessageDigest.getInstance("MD5");
            // 使用指定的字节更新摘要
            mdInst.update(str.getBytes());
            // 获得密文
            byte[] md = mdInst.digest();
            // 把密文转换成十六进制的字符串形式
            StringBuffer hexString = new StringBuffer();
            // 字节数组转换为 十六进制 数
            for (int i = 0; i < md.length; i++) {
                String shaHex = Integer.toHexString(md[i] & 0xFF);
                if (shaHex.length() < 2) {
                    hexString.append(0);
                }
                hexString.append(shaHex);
            }
            return hexString.toString();
        } catch (Exception e) {
            throw new RuntimeException("加密字符串[" + str + "]时遇到异常", e);
        }
    }

    /**
     * MD5加密，返回二进制串
     *
     * @param str：待加密的字符串
     * @return 加密二进制
     */
    public static byte[] MD5ToByteArray(String str) {
        try {
            // 获得MD5摘要算法的 MessageDigest 对象
            MessageDigest mdInst = MessageDigest.getInstance("MD5");
            // 使用指定的字节更新摘要
            mdInst.update(str.getBytes());
            // 获得密文
            byte[] md = mdInst.digest();
            return md;
        } catch (Exception e) {
            throw new RuntimeException("加密字符串[" + str + "]时遇到异常", e);
        }
    }

    /**
     * MD5加密
     *
     * @param bArray：待加密的字符串
     * @return 加密串
     */
    public static String BytesMD5(byte[] bArray) {
        try {
            // 获得MD5摘要算法的 MessageDigest 对象
            MessageDigest mdInst = MessageDigest.getInstance("MD5");
            // 使用指定的字节更新摘要
            mdInst.update(bArray);
            // 获得密文
            byte[] md = mdInst.digest();
            // 把密文转换成十六进制的字符串形式
            StringBuffer hexString = new StringBuffer();
            // 字节数组转换为 十六进制 数
            for (int i = 0; i < md.length; i++) {
                String shaHex = Integer.toHexString(md[i] & 0xFF);
                if (shaHex.length() < 2) {
                    hexString.append(0);
                }
                hexString.append(shaHex);
            }
            return hexString.toString();
        } catch (Exception e) {
            throw new RuntimeException("加密字节[" + bArray + "]时遇到异常", e);
        }
    }

    /**
     * 讲base64字符串解码保存文件
     * @param base64Code base64字符串
     * @param path  目标文件路径
     */
    public static void decodeBase64File(String base64Code,String path){
        byte[] buffer;
        try {
            buffer = new BASE64Decoder().decodeBuffer(base64Code);
            FileOutputStream fos = new FileOutputStream(path);
            fos.write(buffer);
            fos.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

}